import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# EasyScholar API配置
EASYSCHOLAR_API_KEY = os.getenv('EASYSCHOLAR_API_KEY', '')
EASYSCHOLAR_BASE_URL = 'https://www.easyscholar.cc'
EASYSCHOLAR_API_ENDPOINT = '/open/getPublicationRank'

# 筛选配置
# 要保留的分区（默认保留SCI 1区、2区和中科院1区、2区、3区）
KEEP_SCI_QUARTILES = ['Q1', 'Q2']  # 保留SCI 1区、2区
KEEP_CAS_QUARTILES = ['Q1', 'Q2', 'Q3']  # 保留中科院1区、2区、3区

# 是否删除无法获取分区信息的文献
REMOVE_UNKNOWN_QUARTILES = False  # 默认不删除

# 请求配置
REQUEST_TIMEOUT = 30
REQUEST_DELAY = 1  # 请求间隔（秒）

# 输出配置
OUTPUT_SUFFIX = '_filtered' 